from dataclasses import asdict
import json

from via.services.youtube_api import YouTubeAPIClient

client = YouTubeAPIClient()

# Get the video info
video = client.get_video_info("1v8u3ua6BPk")
print("Video details:")
print("\t", video.details.title)
print("\t", video.details.author)
print("\t", video.url)

# Try the matching system
caption_track = video.caption.tracks[0]
print("\nSelecting caption track:")
print("\t", caption_track)

# Get the transcript
transcript = client.get_transcript(caption_track)
print("\nTranscript:")
print(json.dumps(asdict(transcript), indent=4))